#!/bin/bash
set -e  # 出错立即退出

# --------------------------
# 清理残留进程 (防止占用 GPU / spawn 冲突)
# --------------------------
echo "清理残留进程..."
ps -ef | grep "spawn" | grep -v grep | awk '{print $2}' | xargs -r kill -9 || true

# --------------------------
# 配置参数
# --------------------------
NUM_GPUS=4  # 使用的 GPU 数量
VAL_SCRIPT="eval.py"   # 验证脚本路径

# --------------------------
# 启动验证
# --------------------------
echo "开始验证任务..."
echo "使用 GPU 数量: $NUM_GPUS"
echo "使用脚本: $VAL_SCRIPT"

if [ "$NUM_GPUS" -eq 1 ]; then
    # ⚠️ 暂时存在bug！！！！不要使用单卡！！！！
    echo "单 GPU 模式"  
    accelerate launch $VAL_SCRIPT
else
    echo "多 GPU 模式"
    accelerate launch --multi_gpu --num_processes $NUM_GPUS $VAL_SCRIPT
fi

echo "✅ 验证完成"

